CUDA_VISIBLE_DEVICES=1 python3.5 train.py -p helmet_C8 -c 0 --batch_size 8 --lr 1e-5 --num_epochs 500 \
 --data_path /media/gzzn/Data/Datasets/ObjectDetection/helmet/
 --load_weights /media/gzzn/Data/PythonProject/Yet-Another-EfficientDet-Pytorch/pre-train_model/efficientdet-d0.pth \
 --head_only True